package com.tanhua.dubbo.server.api;

import cn.hutool.core.collection.CollUtil;
import cn.hutool.core.util.ObjectUtil;
import com.alibaba.dubbo.config.annotation.Service;
import com.tanhua.dubbo.server.pojo.RecommendUser;
import com.tanhua.dubbo.server.vo.PageInfo;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.domain.PageRequest;
import org.springframework.data.domain.Sort;
import org.springframework.data.mongodb.core.MongoTemplate;
import org.springframework.data.mongodb.core.query.Criteria;
import org.springframework.data.mongodb.core.query.Query;

import java.util.ArrayList;
import java.util.List;

/**
 * 推荐用户业务实现类
 */
@Service(version = "1.0.0")
public class RecommendUserApiImpl implements RecommendUserApi {
    @Autowired
    private MongoTemplate mongoTemplate;
    @Autowired
    private UserLikeApi userLikeApi;

    /**
     * 查询得分最高用户
     *
     * @param userId 用户id
     * @return
     */
    @Override
    public RecommendUser queryWithMaxScore(Long userId) {
        /*
         * 具体查询条件：
         * 1.推荐用户-出入用户id
         * 2.根据得分从高到底排序
         * 3.取第一个【也就是得分最高的用户】
         */
        Query query = Query.query(Criteria.where("toUserId").is(userId))
                .with(Sort.by(Sort.Order.desc("score"))).limit(1);

        // 根据条件查询数据
        RecommendUser recommendUser = this.mongoTemplate.findOne(query, RecommendUser.class);

        return recommendUser;
    }

    /**
     * 获取推荐好友
     * 排序按得分从高到低
     *
     * @param userId   用户id
     * @param pageNum  当前页数
     * @param pageSize 每页条数
     * @return
     */
    @Override
    public PageInfo<RecommendUser> queryPageInfo(Long userId, Integer pageNum, Integer pageSize) {
        /*
         * 分页查询条件
         * 1.page 当前页 【从0页开始，所以减一】
         * 2.size 每页条数
         * 3.根据得分从高到底排序
         */
        PageRequest pageable = PageRequest.of(pageNum - 1, pageSize, Sort.by(Sort.Order.desc("score")));
        /*
         * 查询条件
         * 1.推荐用户-出入用户id
         * 2.分页查询
         *
         */
        Query query = Query.query(Criteria.where("toUserId").is(userId))
                .with(pageable);

        // 根据条件查询数据
        List<RecommendUser> recommendUserList = this.mongoTemplate.find(query, RecommendUser.class);


        return new PageInfo<>(0, pageNum, pageSize, recommendUserList);
    }


    /**
     * 查询推荐好友的缘分值
     *
     * @param userId   好友的id
     * @param toUserId 我的id
     * @return 缘分值
     */
    @Override
    public Double queryScore(Long userId, Long toUserId) {
        Query query = new Query();
        query.addCriteria(Criteria.where("userId").is(userId)
                .and("toUserId").is(toUserId));
        query.fields().include("score"); // 指定查询缘分值字段

        RecommendUser recommendUser = this.mongoTemplate.findOne(query, RecommendUser.class);
        if (ObjectUtil.isNotEmpty(recommendUser)) {
            return recommendUser.getScore();
        }
        return null;
    }


    /**
     * 查询探花列表，查询时需要排除不喜欢列表用户
     *
     * @param userId 用户id
     * @param count  查询个数
     * @return
     */
    @Override
    public List<RecommendUser> queryCardList(Long userId, Integer count) {
        /* 一、收集需要排除的用户id */
        List<Long> excludeUserIdList = new ArrayList<>();
        // 1 排除自己已喜欢用户id
        // 查询自己喜欢用户id
        List<Long> likeUserIdList = this.userLikeApi.queryLikeIdList(userId);
        if (CollUtil.isNotEmpty(likeUserIdList)) {
            excludeUserIdList.addAll(likeUserIdList);
        }

        // 2 排除自己已不喜欢用户id
        // 查询自己不喜欢用户id
        List<Long> notLikeUserList = this.userLikeApi.queryNotLikeList(userId);
        if (CollUtil.isNotEmpty(notLikeUserList)) {
            excludeUserIdList.addAll(notLikeUserList);
        }


        Query query = new Query();
        query.addCriteria(Criteria.where("toUserId").is(userId));
        if (CollUtil.isNotEmpty(excludeUserIdList)) {
            // 加入到查询条件中，排除这些用户
            query.addCriteria(Criteria.where("userId").nin(excludeUserIdList));
        }
        // 分页 根据缘分值降序排序
        PageRequest pageable = PageRequest.of(0, count, Sort.by(Sort.Order.desc("score")));
        query.with(pageable);

        List<RecommendUser> recommendUserList = this.mongoTemplate.find(query, RecommendUser.class);
        return recommendUserList;
    }
}
